算法入门系列2 - 在水一方

在第一次的公开课中,我们讲到了穷举法。穷举法也被称为暴力搜索法,今天我们要讲的回溯法就是暴力搜索法的一种。接下来我们讲到的很多算法跟“递归”这个概念有或多或少的关系,所以我们先说说“递归”。

现实中的递归

从前有座山,山里有座庙,庙里有个老和尚,正在给小和尚讲故事呢!故事是什么呢?从前有座山,山里有座庙,庙里有个老和尚,正在给小和尚讲故事呢!故事是什么呢?从前有座山,山里有座庙,庙里有个老和尚,正在给小和尚讲故事呢!故事是什么呢?……

野比大雄在房间里,用时光电视看着未来的情况。电视画面中,野比大雄在房间里,用时光电视看着未来的情况。电视画面中,野比大雄在房间里,用时光电视看着未来的情况……

阶乘的递归定义:0! = 1n!=n*(n-1)! ,使用被定义对象的自身来为其下定义称为递归定义。

德罗斯特效应是递归的一种视觉形式。图中女性手持的物体中有一幅她本人手持同一物体的小图片,进而小图片中还有更小的一幅她手持同一物体的图片……

算法入门系列2 - 在水一方 - 图1

递归的应用

在程序中,一个函数如果直接或者间接的调用了自身,我们就称之为递归函数。

写递归函数有两个要点:

  1. 收敛条件 - 什么时候结束递归。
  2. 递归公式 - 每一项与前一项(前N项)的关系。

例子1:求阶乘。

  1. def fac(num):
  2. if num == 0:
  3. return 1
  4. return num * fac(num - 1)

Python对递归的深度加以了限制(默认1000层函数调用),如果想突破这个限制,可以使用下面的方法。

  1. import sys
  2. sys.setrecursionlimit(10000)

例子2:爬楼梯 - 楼梯有n个台阶,一步可以走1阶、2阶或3阶,走完n个台阶共有多少种不同的走法。

  1. def climb(num):
  2. if num == 0:
  3. return 1
  4. elif num < 0:
  5. return 0
  6. return climb(num - 1) + climb(num - 2) + climb(num - 3)

注意:上面的递归函数性能会非常的差,因为时间复杂度是几何级数级的。

优化后的代码。

  1. from functools import lru_cache
  2. @lru_cache()
  3. def climb(num):
  4. if num == 0:
  5. return 1
  6. elif num < 0:
  7. return 0
  8. return climb(num - 1) + climb(num - 2) + climb(num - 3)

不使用的递归的代码。

  1. def climb(num):
  2. a, b, c = 1, 2, 4
  3. for _ in range(num - 1):
  4. a, b, c = b, c, a + b + c
  5. return a

重点:有更好的办法的时候,请不要考虑递归。

回溯法

回溯法暴力搜索法中的一种。对于某些计算问题而言,回溯法是一种可以找出所有(或一部分)解的一般性算法,尤其适用于约束满足问题(在解决约束满足问题时,我们逐步构造更多的候选解,并且在确定某一部分候选解不可能补全成正确解之后放弃继续搜索这个部分候选解本身及其可以拓展出的子候选解,转而测试其他的部分候选解)。

经典案例

例子1:迷宫寻路

算法入门系列2 - 在水一方 - 图2

  1. """
  2. 迷宫寻路
  3. """
  4. import random
  5. import sys
  6. WALL = -1
  7. ROAD = 0
  8. ROWS = 10
  9. COLS = 10
  10. def find_way(maze, i=0, j=0, step=1):
  11. """走迷宫"""
  12. if 0 <= i < ROWS and 0 <= j < COLS and maze[i][j] == 0:
  13. maze[i][j] = step
  14. if i == ROWS - 1 and j == COLS - 1:
  15. print('=' * 20)
  16. display(maze)
  17. sys.exit(0)
  18. find_way(maze, i + 1, j, step + 1)
  19. find_way(maze, i, j + 1, step + 1)
  20. find_way(maze, i - 1, j, step + 1)
  21. find_way(maze, i, j - 1, step + 1)
  22. maze[i][j] = ROAD
  23. def reset(maze):
  24. """重置迷宫"""
  25. for i in range(ROWS):
  26. for j in range(COLS):
  27. num = random.randint(1, 10)
  28. maze[i][j] = WALL if num > 7 else ROAD
  29. maze[0][0] = maze[ROWS - 1][COLS - 1] = ROAD
  30. def display(maze):
  31. """显示迷宫"""
  32. for row in maze:
  33. for col in row:
  34. if col == -1:
  35. print('■', end=' ')
  36. elif col == 0:
  37. print('□', end=' ')
  38. else:
  39. print(f'{col}'.ljust(2), end='')
  40. print()
  41. def main():
  42. """主函数"""
  43. maze = [[0] * COLS for _ in range(ROWS)]
  44. reset(maze)
  45. display(maze)
  46. find_way(maze)
  47. print('没有出路!!!')
  48. if __name__ == '__main__':
  49. main()

说明:上面的代码用随机放置围墙的方式来生成迷宫,更好的生成迷宫的方式请参考《简单的使用回溯法生成 Tile Based 迷宫》一文。

例子2:骑士巡逻 - 国际象棋中的骑士(马),按照骑士的移动规则走遍整个棋盘的每一个方格,而且每个方格只能够经过一次。

算法入门系列2 - 在水一方 - 图3

  1. """
  2. 骑士巡逻
  3. """
  4. import sys
  5. SIZE = 8
  6. def display(board):
  7. """显示棋盘"""
  8. for row in board:
  9. for col in row:
  10. print(f'{col}'.rjust(2, '0'), end=' ')
  11. print()
  12. def patrol(board, i=0, j=0, step=1):
  13. """巡逻"""
  14. if 0 <= i < SIZE and 0 <= j < SIZE and board[i][j] == 0:
  15. board[i][j] = step
  16. if step == SIZE * SIZE:
  17. display(board)
  18. sys.exit(0)
  19. patrol(board, i + 1, j + 2, step + 1)
  20. patrol(board, i + 2, j + 1, step + 1)
  21. patrol(board, i + 2, j - 1, step + 1)
  22. patrol(board, i + 1, j - 2, step + 1)
  23. patrol(board, i - 1, j - 2, step + 1)
  24. patrol(board, i - 2, j - 1, step + 1)
  25. patrol(board, i - 2, j + 1, step + 1)
  26. patrol(board, i - 1, j + 2, step + 1)
  27. board[i][j] = 0
  28. def main():
  29. """主函数"""
  30. board = [[0] * SIZE for _ in range(SIZE)]
  31. patrol(board)
  32. if __name__ == '__main__':
  33. main()

例子3:八皇后 - 如何能够在8×8的国际象棋棋盘上放置八个皇后,使得任何一个皇后都无法直接吃掉其他的皇后?为了达到此目的,任两个皇后都不能处于同一条横行、纵行或斜线上。

算法入门系列2 - 在水一方 - 图4

说明:这个问题太经典了,网上有大把的答案,留给大家自己搞定。